﻿//The MIT License

//Copyright (c) 2009 Lluis Sanchez, Nick Darnell

//Permission is hereby granted, free of charge, to any person obtaining a copy
//of this software and associated documentation files (the "Software"), to deal
//in the Software without restriction, including without limitation the rights
//to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
//copies of the Software, and to permit persons to whom the Software is
//furnished to do so, subject to the following conditions:

//The above copyright notice and this permission notice shall be included in
//all copies or substantial portions of the Software.

//THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
//IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
//FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
//AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
//LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
//OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
//THE SOFTWARE.

using System;
using System.Collections.Generic;
using System.Reflection.Emit;
using System.Text;

using NUnit.Framework;
using System.Collections;

namespace Polymorph.Test
{
    [TestFixture]
    public class ForeachTestFixture
    {
        [Test]
        public void TestForeachArray()
        {
            CodeDynamicMethod method = 
                new CodeDynamicMethod("Foreach", typeof(void), Type.EmptyTypes, true);
            CodeBuilder cb = method.CodeBuilder;

            CodeLocalReference local = cb.DeclareVariable<int[]>(Exp.NewArray<int>(5));

            CodeValueReference item;
            cb.Foreach<int>(out item, local);
            cb += ExpConsole.WriteLine("Loop");
            cb.EndForeach();

            Console.WriteLine(cb.PrintCode());

            DynamicMethod dm = method.Generate();
            dm.Invoke(null, new object[] { });
        }

        [Test]
        public void TestForeachIEnumerable()
        {
            CodeDynamicMethod method =
                new CodeDynamicMethod("Foreach", typeof(void), Type.EmptyTypes, true);
            CodeBuilder cb = method.CodeBuilder;

            CodeLocalReference local = cb.DeclareVariable<SampleContainer>(Exp.New<SampleContainer>());

            CodeValueReference item;
            cb.Foreach<object>(out item, local);
            cb += ExpConsole.WriteLine("Loop");
            cb.EndForeach();

            Console.WriteLine(cb.PrintCode());

            DynamicMethod dm = method.Generate();
            dm.Invoke(null, new object[] { });
        }

        [Test]
        public void TestForeachIEnumerableGeneric()
        {
            CodeDynamicMethod method =
                new CodeDynamicMethod("Foreach", typeof(void), Type.EmptyTypes, true);
            CodeBuilder cb = method.CodeBuilder;

            CodeLocalReference local = 
                cb.DeclareVariable<SampleContainerGeneric>(Exp.New<SampleContainerGeneric>());

            CodeValueReference item;
            cb.Foreach<int>(out item, local);
            cb += ExpConsole.WriteLine("Loop");
            cb.EndForeach();

            Console.WriteLine(cb.PrintCode());

            DynamicMethod dm = method.Generate();
            dm.Invoke(null, new object[] { });
        }

        private class SampleContainer : IEnumerable
        {
            private List<int> m_container = new List<int>(5);

            #region IEnumerable Members

            public IEnumerator GetEnumerator()
            {
                return m_container.GetEnumerator();
            }

            #endregion
        }

        private class SampleContainerGeneric : IEnumerable<int>
        {
            private List<int> m_container = new List<int>(5);

            #region IEnumerable<int> Members

            public IEnumerator<int> GetEnumerator()
            {
                return m_container.GetEnumerator();
            }

            #endregion

            #region IEnumerable Members

            IEnumerator IEnumerable.GetEnumerator()
            {
                throw new NotImplementedException();
            }

            #endregion
        }
    }
}
